package stru1.day10;

import java.util.ArrayList;
import java.util.List;

/**
 * 145. 二叉树的后序遍历
 */
public class Solution3 {
    public static void main(String[] args) {
        System.out.println(postorderTraversal(new TreeNode(1, null, new TreeNode(2, new TreeNode(3), null))));
        System.out.println(postorderTraversal(null));
        System.out.println(postorderTraversal(new TreeNode(1)));
    }

    /**
     * Morris遍历
     * Morris遍历有另外一个学名叫做：线索二叉树。如果在遍历一棵树时严令禁止修改树的结构，那么Morris遍历就用不了。
     * <p>
     * --开始时cur来到根节点位置：
     * --如果cur有左孩子，找到左子树上最右的节点mostRight
     * ----如果mostRight的右指针指向null，让其指向cur，然后cur向左移动（cur = cur.left）
     * ----如果mostRight的右指针指向cur，让其指向null，然后cur向右移动（cur = cur.right）
     * --如果cur没有左孩子，cur向右移动（cur = cur.right）
     * --cur为空时遍历停止
     * <p>
     * 另外两种解法：递归，迭代
     * 时间复杂度 O(n)
     * 空间复杂度 O(1)
     */
    public static List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        if (root == null) {
            return res;
        }

        TreeNode p1 = root, p2;

        while (p1 != null) {
            p2 = p1.left;
            if (p2 != null) {
                while (p2.right != null && p2.right != p1) {
                    p2 = p2.right;
                }
                if (p2.right == null) {
                    p2.right = p1;
                    p1 = p1.left;
                    continue;
                } else {
                    p2.right = null;
                    addPath(res, p1.left);
                }
            }
            p1 = p1.right;
        }
        addPath(res, root);
        return res;
    }

    public static void addPath(List<Integer> res, TreeNode node) {
        int count = 0;
        while (node != null) {
            ++count;
            res.add(node.val);
            node = node.right;
        }
        int left = res.size() - count, right = res.size() - 1;
        while (left < right) {
            int temp = res.get(left);
            res.set(left, res.get(right));
            res.set(right, temp);
            left++;
            right--;
        }
    }
}


